package main

import (
	"fmt"
	"gorm.io/driver/mysql"
	"gorm.io/gen"
	"gorm.io/gorm"
)

func main() {
	//HOST: '8.137.157.84'
	//USER: 'root'
	//PWD: 'dc494235908'
	//PORT: '30036'
	//Database: 'database_serve
	HOST := "8.137.157.84"
	USER := "root"
	PWD := "dc494235908"
	PORT := "30036"
	Database := "database_serve"
	//DNS :=${USER}":"${PWD}"@tcp("${HOST}":"${PORT}")/"${USER}"?charset=utf8mb4&parseTime=True&loc=Local"
	DNS := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=Local", USER, PWD, HOST, PORT, Database)
	g := gen.NewGenerator(gen.Config{
		//  设置输出路径
		OutPath: "./query",

		// WithDefaultQuery 生成默认查询结构体(作为全局变量使用), 即`Q`结构体和其字段(各表模型)
		// WithoutContext 生成没有context调用限制的代码供查询
		// WithQueryInterface 生成interface形式的查询代码(可导出), 如`Where()`方法返回的就是一个可导出的接口类型
		Mode: gen.WithDefaultQuery | gen.WithQueryInterface,

		// 表字段可为 null 值时, 对应结体字段使用指针类型
		FieldNullable: false, // generate pointer when field is nullable

		// 表字段默认值与模型结构体字段零值不一致的字段, 在插入数据时需要赋值该字段值为零值的, 结构体字段须是指针类型才能成功, 即`FieldCoverable:true`配置下生成的结构体字段.
		// 因为在插入时遇到字段为零值的会被GORM赋予默认值. 如字段`age`表默认值为10, 即使你显式设置为0最后也会被GORM设为10提交.
		// 如果该字段没有上面提到的插入时赋零值的特殊需要, 则字段为非指针类型使用起来会比较方便.
		FieldCoverable: false, // generate pointer when field has default value, to fix problem zero value cannot be assign: https://gorm.io/docs/create.html#Default-Values

		// 模型结构体字段的数字类型的符号表示是否与表字段的一致, `false`指示都用有符号类型
		FieldSignable: false, // detect integer field's unsigned type, adjust generated data type
		// 生成 gorm 标签的字段索引属性
		FieldWithIndexTag: false, // generate with gorm index tag
		// 生成 gorm 标签的字段类型属性
		FieldWithTypeTag: true, // generate with gorm column type tag
	})
	//  建立数据库连接
	gormDb, _ := gorm.Open(mysql.Open(DNS))
	g.UseDB(gormDb) // 选择数据库连接

	var tables []string
	result := gormDb.Raw("SHOW TABLES").Scan(&tables)
	if result.Error != nil {
		panic("Failed to fetch tables: " + result.Error.Error())
	}
	fmt.Println("Tables in database:")
	for _, table := range tables {
		fmt.Println(table)
	}
	dataMap := map[string]func(detailType gorm.ColumnType) (dataType string){
		//"datetime": func(detailType gorm.ColumnType) (dataType string) { return "*utils.LocalTime" },
	}

	g.WithDataTypeMap(dataMap)
	g.ApplyBasic(

		// 从当前数据库的所有表生成结构
		g.GenerateAllTable()...,
	)
	// 生成代码
	g.Execute()
}
